package org.sky.algorithm.sort;

import java.util.Arrays;

/**
 * 合并排序
 *
 * @author fog
 * @date 2021/11/19
 */

public class MergeSort {

    public static void main(String[] args) {

        int[] arr = new int[]{1, 2, 3, 40, 2, 6, 7, 8, 9};

        System.out.println(Arrays.toString(arr));
        mergeSort(arr, 0, arr.length - 1);
        System.out.println(Arrays.toString(arr));

    }

    /**
     * 归并排序
     *
     * @param arr  数组
     * @param low  下标为0
     * @param high 数组长度减一
     */
    public static void mergeSort(int[] arr, int low, int high) {
        int middle = (low + high) / 2;
        if (low < high) {
            // 处理左边数组
            mergeSort(arr, low, middle);
            // 处理右边数组
            mergeSort(arr, middle + 1, high);
            // 归并
            merge(arr, low, middle, high);
        }
    }

    /**
     * 归并
     *
     * @param arr    数组
     * @param low    下标为 0；
     * @param middle 第一个数组长度减一
     * @param high   第二个数组长度减一
     */
    public static void merge(int[] arr, int low, int middle, int high) {

        // 创建临时数组
        int[] temp = new int[high - low + 1];

        // 记录第一个数组中需要遍历的下标
        int i = low;
        // 记录第二个数组需要遍历的下标
        int j = middle + 1;
        // 记录临时存放的数组下标
        int index = 0;

        while (i <= middle && j <= high) {

            if (arr[i] <= arr[j]) {
                temp[index] = arr[i];
                i++;
            } else {
                temp[index] = arr[j];
                j++;
            }

            index++;

        }

        // 处理多余数组
        while (i <= middle) {
            temp[index] = arr[i];
            i++;
            index++;
        }
        while (j <= high) {
            temp[index] = arr[j];
            j++;
            index++;
        }

        // 把临时数组赋给原数组
        for (int k = 0; k < temp.length; k++) {
            arr[k + low] = temp[k];
        }

    }

}
